package network;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Scanner;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class TcpEchoServer {
    public ServerSocket socket = null;

    public TcpEchoServer(int port) throws IOException {
        socket = new ServerSocket(port);
    }

    public void start() throws IOException {
        System.out.println("服务器启动");
        //因为 TCP是有连接的，所以我们需要先处理一下连接
        //这里的线程个数是不能进行固定的
        ExecutorService executorService = Executors.newCachedThreadPool();
        while(true) {
            // tcp 来说, 需要先处理客户端发来的连接.
            // 通过读写 clientSocket, 和客户端进行通信.
            // 如果没有客户端发起连接, 此时 accept 就会阻塞.
            //这里的 Socket 不是跟随进程的，所以在处理完一次请求后，我们要对于进行关闭
            Socket clientSocket = socket.accept();

            //这里会出现多个客户端对其服务器进行请求操作，所以会出现多线程的问题，所以我们这里需要进行解决
            //但是使用这种方法的会 频繁的创建销毁线程，我们可以使用线程池进行处理
//            Thread t = new Thread(() -> {
//                processConnection(clientSocket);
//            });

            executorService.submit(() -> {
                processConnection(clientSocket);
            });
        }
    }

    // 处理一个客户端的连接.
    // 可能要涉及到多个客户端的请求和响应.
    private void processConnection(Socket clientSocket) {
        System.out.printf("[%s : %d] 客户端上线!\n",clientSocket.getInetAddress(),clientSocket.getPort());
        //这里的输入输出是使用 字节流来完成的
        try (InputStream inputStream = clientSocket.getInputStream();
             OutputStream outputStream = clientSocket.getOutputStream()) {
            //对于InputStream进行套一层的操作，这样就可以使用scanner接收字符串了
            Scanner scanner = new Scanner(inputStream);
            //对于OutputStream进行套一层的操作，这样就可以使用writer发送字符串了
            PrintWriter writer = new PrintWriter(outputStream);
            while(true) {
                // 1、读取客户端的数据并且进行解析
                //  对于这里的接收操作，可以使用 inputStream.read() 进行接收，但是需要进行转换成 字符串
                //  所以我们可以使用 scanner 进行读取接收
                if (!scanner.hasNext()) {
                    System.out.printf("[%s:%d] 客户端下线!\n", clientSocket.getInetAddress(), clientSocket.getPort());
                    break;
                }
                String request = scanner.next();

                // 2、根究解析的数据，计算出其回响值
                String response = process(request);

                // 3、给客户端返回其回响值
                //  这里的 PrintWriter 的操作呢会发送到 “缓冲区” 中，所以每次的发送，我们都需要对其进行刷新，强行发送出去
                writer.println(response);
                writer.flush();

                //打印日志
                System.out.printf("[%s:%d] req: %s, resp: %s\n", clientSocket.getInetAddress(), clientSocket.getPort(),
                        request, response);
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        } finally {
            try {
                clientSocket.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }
    }

    private String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        TcpEchoServer tcpEchoServer = new TcpEchoServer(9090);
        tcpEchoServer.start();
    }
}
